package com.lagou.rpc.consumer.proxy;


import com.alibaba.fastjson.JSON;
import com.lagou.rpc.common.RpcRequest;
import com.lagou.rpc.common.RpcResponse;
import com.lagou.rpc.consumer.client.RpcClient;
import com.lagou.rpc.consumer.client.RpcClientCluster;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.UUID;

@Component
@ConfigurationProperties(prefix = "rpc")
public class RemoteServiceInvocationHandler implements InvocationHandler, InitializingBean {
    private String serverList;

    public void setServerList(String serverList) {
        this.serverList = serverList;
    }

    @Autowired
    private RpcClientCluster clientCluster;

    @Override
    public void afterPropertiesSet() throws Exception {
        //clientCluster.init(serverList);
        clientCluster.init();
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        //1.封装request请求对象
        RpcRequest rpcRequest = new RpcRequest();
        rpcRequest.setRequestId(UUID.randomUUID().toString());
        rpcRequest.setClassName(method.getDeclaringClass().getName());
        rpcRequest.setMethodName(method.getName());
        rpcRequest.setParameterTypes(method.getParameterTypes());
        rpcRequest.setParameters(args);
        //2.创建RpcClient对象
        //RpcClient rpcClient = new RpcClient("127.0.0.1", 8899);
        //使用RpcClientCluster实现负载均衡
        RpcClient rpcClient = clientCluster.getRpcClient();
        try {
            //3.发送消息
            Object responseMsg = rpcClient.send(JSON.toJSONString(rpcRequest));
            RpcResponse rpcResponse = JSON.parseObject(responseMsg.toString(), RpcResponse.class);
            if (rpcResponse.getError() != null) {
                throw new RuntimeException(rpcResponse.getError());
            }
            //4.返回结果
            Object result = rpcResponse.getResult();
            return JSON.parseObject(result.toString(), method.getReturnType());
        } catch (Exception e) {
            throw e;
        } finally {
            //rpcClient.close();
        }
    }

}
